package org.monkey.common.gateway.controller;

import java.util.Optional;
import org.monkey.common.gateway.config.SwaggerResourceConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Mono;
import springfox.documentation.swagger.web.SecurityConfiguration;
import springfox.documentation.swagger.web.SecurityConfigurationBuilder;
import springfox.documentation.swagger.web.UiConfiguration;
import springfox.documentation.swagger.web.UiConfigurationBuilder;

/**
 * 自定义Swagger的各个配置节点
 */
@RestController
public class SwaggerController {

  @Autowired(required = false)
  private SecurityConfiguration securityConfiguration;

  @Autowired(required = false)
  private UiConfiguration uiConfiguration;

  @Autowired
  private SwaggerResourceConfig swaggerResourceConfig;


  /**
   * Swagger安全配置，支持oauth和apiKey设置
   */
  @GetMapping("/swagger-resources/configuration/security")
  public Mono<ResponseEntity<SecurityConfiguration>> securityConfiguration() {
    return Mono.just(new ResponseEntity<>(
        Optional.ofNullable(securityConfiguration)
            .orElse(SecurityConfigurationBuilder.builder().build()), HttpStatus.OK));
  }

  /**
   * Swagger UI配置
   */
  @GetMapping("/swagger-resources/configuration/ui")
  public Mono<ResponseEntity<UiConfiguration>> uiConfiguration() {
    return Mono.just(new ResponseEntity<>(
        Optional.ofNullable(uiConfiguration).orElse(UiConfigurationBuilder.builder().build()),
        HttpStatus.OK));
  }

  /**
   * Swagger资源配置，微服务中这各个服务的api-docs信息
   */
  @GetMapping("/swagger-resources")
  public Mono<ResponseEntity> swaggerResources() {
    return Mono.just((new ResponseEntity<>(swaggerResourceConfig.get(), HttpStatus.OK)));
  }
}
